from ..builder import DETECTORS
from .single_stage import SingleStageDetector


@DETECTORS.register_module()
class RepPointsDetector(SingleStageDetector):
    """RepPoints: Point Set Representation for Object Detection.

    This detector is the implementation of:
    - RepPoints detector (https://arxiv.org/pdf/1904.11490)
    """

    def __init__(
        self, backbone, neck, bbox_head, train_cfg=None, test_cfg=None, pretrained=None
    ):
        super(RepPointsDetector, self).__init__(
            backbone, neck, bbox_head, train_cfg, test_cfg, pretrained
        )
